export dataset_name=/home/user4/workspace/tm-benchmark/data

# prot_bert prot_bert_bfd prot_t5_xl_uniref50 prot_t5_xl_bfd
CUDA_VISIBLE_DEVICES=6 python src/ProtTrans/predict.py \
    --dataset_input $dataset_name \
    --model_location prot_bert prot_bert_bfd prot_t5_xl_uniref50 prot_t5_xl_bfd \
    --mutation_col mutant